# 导入工具包
import fasttext
from ft01_config import Config

#1、导入配置文件
conf = Config()

# 2、模型训练
model = fasttext.train_supervised(
    #无监督情况下 wordNgrams等同于word2vec
    input='./final_data/train_fastText_jieba.txt',
)

#3、模型保存
path = conf.ft_model_save_path
model.save_model(path+"./fastText_jieba_default.bin")

#4、模型预测
print(model.predict("名师详解考研复试英语听力考策略"))

# 5、模型词表查看
# print(model.words)
# print(len(model.words))

# 6、查看模型子词，上述训练未开启子词，所以这里查到还是词本身
print(model.get_subwords('你好'))

# 7、模型测试
print("模型测试验证评估开始...")
res = model.test('./final_data/test_fastText_jieba.txt')
print(res)

